broadcasting semantics

https://pytorch.org/docs/stable/notes/broadcasting.html#broadcasting-semantics